Machine learning-based prediction model